import torch
from transformers import AutoModel

mode_path = 'E:\\ai\\huggingface\\models\\TinyBERT_General_4L_312D'
model = AutoModel.from_pretrained(mode_path, local_files_only=True)
(result, label) = model(torch.rand(1, 3, 224, 224))
print(result, label)

